package Elementary_algorithm.Other;

import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/*
杨辉三角
给定一个非负整数numRows，生成「杨辉三角」的前numRows行。
在「杨辉三角」中，每个数是它左上方和右上方的数的和。
示例 1:
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
示例2:
输入: numRows = 1
输出: [[1]]
作者：LeetCode
链接：https://leetcode.cn/leetbook/read/array-and-string/cuj3m/
 */
public class _04杨辉三角 {
    @Test
    public void test() {
        System.out.println(generate(5));
    }

    //数学 + 模拟
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 1; i <= numRows; i++) {
            List<Integer> list = new ArrayList<>();
            for (int j = 0; j < i; j++) {
                if (j == 0 || j == i - 1) {
                    list.add(1);
                    continue;
                }
                list.add(res.get(i - 2).get(j - 1) + res.get(i - 2).get(j));
            }
            res.add(list);
        }
        return res;
    }

    //官方写法：方法一：数学
    /*
    作者：力扣官方题解
    链接：https://leetcode.cn/problems/pascals-triangle/solutions/510638/yang-hui-san-jiao-by-leetcode-solution-lew9/
     */
    class Solution {
        public List<List<Integer>> generate(int numRows) {
            List<List<Integer>> ret = new ArrayList<List<Integer>>();
            for (int i = 0; i < numRows; ++i) {
                List<Integer> row = new ArrayList<Integer>();
                for (int j = 0; j <= i; ++j) {
                    if (j == 0 || j == i) {
                        row.add(1);
                    } else {
                        row.add(ret.get(i - 1).get(j - 1) + ret.get(i - 1).get(j));
                    }
                }
                ret.add(row);
            }
            return ret;
        }
    }

}
